from llama_index.core.base.llms.types import ChatMessage
from workflows import Workflow, step
from workflows.events import Event, StartEvent


class InputEvent(Event):
    input:list[ChatMessage]

class MyWorkflow(Workflow):
    @step
    async def first_step(self,ev:StartEvent)->InputEvent:
        return None
